#ifndef __CUDA_UTIL_H__
#define __CUDA_UTIL_H__

#include "pch.h"

void YUV420P_to_NV12(
    const uint8_t *u_plane, const uint8_t *v_plane, uint8_t *uv_plane, int width, int height, int u_pitch, int v_pitch, int uv_pitch, cudaStream_t stream);

void BGR_to_NV12(const uint8_t *src, uint8_t *y_plane, uint8_t *uv_plane, int src_step, int y_step, int uv_step, int width, int height, cudaStream_t stream);

#endif // __CUDA_UTIL_H__